package com.thekernel.youdaoyun.util;

import sun.misc.BASE64Decoder;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import java.security.Key;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Random;
import java.util.UUID;

/**
 * @Author TheKernel
 * @Date 2019/9/16 2:06 下午
 * @Version 1.0
 * @Description 密码、验证码等工具类
 */
public class PasswordUtils {

    private static char[] codeSequence = {
            'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U',
            'V', 'W', 'X', 'Y', 'Z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'd', 'e', 'f', 'g',
            'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z'
    };
    private static Key key;
    private static String CHARSET = "UTF-8";
    private static String ALGORITHM = "DES";

    static {
        try {
            // 生成DES算法对象
            KeyGenerator generator = KeyGenerator.getInstance(ALGORITHM);
            //运用SHA1安全策略
            SecureRandom secureRandom = SecureRandom.getInstance("SHA1PRNG");
            //设置秘钥key
            String KEY_STR = "TheKernel";
            //设置上密钥种子
            secureRandom.setSeed(KEY_STR.getBytes());
            //初始化基于SHA1的算法对象
            generator.init(secureRandom);
            //生成密钥对象
            key = generator.generateKey();
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        }
    }

    /**
     * 对登录密码进行加密
     *
     * @param password 原密码
     * @return 加密后的密码
     */
    public static String encrypt(String password) {
        // return DigestUtils.md5DigestAsHex(password.getBytes());
        return password;
    }

    /**
     * 对文件名进行加密
     *
     * @return 加密后的文件名
     */
    public static String encryptFilename() {
        // 根据时间随机生成文件名
//        String RandomFilename = "";
//        Random rand = new Random();
//        int random = rand.nextInt();
//
//        Calendar calCurrent = Calendar.getInstance();
//        int intDay = calCurrent.get(Calendar.DATE);
//        int intMonth = calCurrent.get(Calendar.MONTH) + 1;
//        int intYear = calCurrent.get(Calendar.YEAR);
//        String now = intYear + "_" + intMonth + "_" + intDay + "_";
//        RandomFilename = now + ((random > 0) ? random : ((-1) * random));
//
//        return RandomFilename;

//        // 基于BASE64编码，接收byte[]并转换成String
//        BASE64Encoder base64Encoder = new BASE64Encoder();
//        try {
//            // 按UTF8编码
//            byte[] bytes = filename.getBytes(CHARSET);
//            // 获取加密对象
//            Cipher cipher = Cipher.getInstance(ALGORITHM);
//            // 初始化密码信息
//            cipher.init(Cipher.ENCRYPT_MODE, key);
//            // 加密
//            byte[] code = cipher.doFinal(bytes);
//            // byte[] to encode 好的 String 并返回
//            return base64Encoder.encode(code);
//        } catch (Exception e) {
//            throw new RuntimeException(e);
//        }

        return UUID.randomUUID().toString().replace("-", "");
    }

    /**
     * 对密码进行解密为文件名
     *
     * @param code 文件名密码
     * @return 文件名
     */
    public static String decryptFilename(String code) {
        // 基于BASE64编码，接收byte[]并转换成String
        BASE64Decoder base64decoder = new BASE64Decoder();
        try {
            // 将字符串decode成byte[]
            byte[] bytes = base64decoder.decodeBuffer(code);
            // 获取解密对象
            Cipher cipher = Cipher.getInstance(ALGORITHM);
            // 初始化解密信息
            cipher.init(Cipher.DECRYPT_MODE, key);
            // 解密
            byte[] filename = cipher.doFinal(bytes);
            // 返回解密之后的信息
            return new String(filename, CHARSET);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 生成验证码(默认从 26个字母(大小写) + 10个数字 中随机)
     *
     * @return 4 位数验证码
     */
    public static String generateVerificationCode() {
        StringBuilder code = new StringBuilder();
        Random random = new Random();
        for (int i = 0; i < 4; i++) {
            code.append(codeSequence[random.nextInt(codeSequence.length - 1)]);
        }

        return code.toString();
    }

    /**
     * 根据用户提供的材料以及长度生成对应的验证码
     *
     * @param source 生成验证码的材料
     * @param length 需要生成验证码的长度
     * @return length 长度的验证码
     */
    public static String generateVerificationCode(char[] source, int length) {
        StringBuilder code = new StringBuilder();
        Random random = new Random();
        for (int i = 0; i < length; i++) {
            code.append(source[random.nextInt(source.length - 1)]);
        }

        return code.toString();
    }

}
